fix: retrieving data from niewiem scrip[t

Backend_correct/app.py

@@ -16,8 +16,8 @@ def recommended_mediators():
 
     input = data.get('request_data', {})
     
-    tmp = os.system('python3 ..\statystyki\load_data.py -t 1 -k 25000 -b True')
+    result = calc_stats("2", 25000, True)
-    print(tmp)
+    print(result)
 
     top_5 = {
         "response_type": "recommended_mediators",
@@ -45,10 +45,88 @@ def recommended_mediators():
             "ai_rating": 90,
             "user_rating": 99,
             "number_of_opinions": 5
-            }], tmp]
+            }]]
         }
 
-    return jsonify(tmp)
+    return jsonify(top_5)
 
+
+def calc_stats(typ,kwota,biegly):
+    mapka = {'1':'cywilnej','2':'górniczej','3':'gospodarczej','4':'prawa pracy & ubezpieczeń'}
+    koszt_bieglego= 1789.42
+    koszt = 0
+    danePath = 'dane.xlsx'
+    if kwota < 100000 and typ != '4':
+        df = pd.read_excel(danePath,sheet_name='rejon')
+        mask = df['RODZAJ'] == int(typ)
+        liczba_miesiecy = df[mask]['mean']
+        procent = (1 - df[mask]['procent do 12 miesięcy']) * 100
+        koszt_sadu = 0
+        koszt_adwokata = 0
+        if kwota <= 500:
+            koszt_sadu = 30
+            koszt_adwokata = 90
+        elif kwota > 500 and kwota <= 1500:
+            koszt_sadu = 100
+            koszt_adwokata = 270
+        elif kwota > 1500 and kwota <= 4000:
+            koszt_sadu = 200
+        elif kwota > 4000 and kwota <= 7500:
+            koszt_sadu = 400
+        elif kwota > 7500 and kwota <= 10000:
+            koszt_sadu = 500    
+        elif kwota > 10000 and kwota <= 15000:
+            koszt_sadu = 750
+        elif kwota > 15000 and kwota <= 20000:
+            koszt_sadu = 1000
+        elif kwota > 20000:
+            koszt_sadu = kwota * 0.05
+            if koszt_sadu > 20000:
+                koszt_sadu = 20000
+
+        if kwota > 1500 and kwota <= 5000:
+            koszt_adwokata = 900
+        elif kwota > 5000 and kwota <= 10000:
+            koszt_adwokata = 1800
+        elif kwota > 10000 and kwota <= 50000:
+            koszt_adwokata = 3600    
+        elif kwota > 50000 and kwota <= 100000:
+            koszt_adwokata = 5400
+
+        
+        if biegly == 'True':
+            koszt = koszt_sadu + koszt_adwokata + koszt_bieglego
+        else:
+            koszt = koszt_sadu + koszt_adwokata
+        print(f"Średni czas trwania rozprawy typu {mapka[typ]} wynosi {round(liczba_miesiecy,0).to_string(index=False)} miesięcy, a {procent.to_string(index=False)}% spraw trwa dłuzej niz rok, jej minimalny koszt wyniesie {koszt}")
+    
+    elif kwota > 100000:
+        df = pd.read_excel(danePath,sheet_name='okreg')
+        mask = df['RODZAJ'] == int(typ)
+        liczba_miesiecy = df[mask]['mean']
+        procent = (1 - df[mask]['procent do 12 miesięcy']) * 100
+
+        if kwota <= 200000:
+            koszt_adwokata = 5400
+        elif kwota > 200000 and kwota <= 2000000:
+            koszt_adwokata = 10800
+        elif kwota > 2000000 and kwota <= 5000000:
+            koszt_adwokata = 15000
+        elif kwota > 500000:
+            koszt_adwokata = 25000
+        
+        koszt_sadu = kwota * 0.05
+        if koszt_sadu > 20000:
+            koszt_sadu = 20000
+        
+        if biegly == 'True':
+            koszt = koszt_sadu + koszt_adwokata + koszt_bieglego
+        else:
+            koszt = koszt_sadu + koszt_adwokata
+
+        return_string = f"Średni czas trwania rozprawy typu {mapka[TYP]} wynosi {round(liczba_miesiecy,0).to_string(index=False)} miesięcy, a {procent.to_string(index=False)}% spraw trwa krócej niz rok, a jej minimalny koszt wynosi {koszt}"     
+        print(return_string)
+
+        return return_string
 if __name__ == '__main__':
     app.run(debug=True)