Update app.py

app.py

@@ -4,26 +4,33 @@ import numpy as np
 import json
 import requests
 import gradio as gr
-
+import logging
 
 bert_tokenizer = BertTokenizer.from_pretrained('MultiTokenizer_ep10')
 bert_model = TFBertForSequenceClassification.from_pretrained('MultiModel_ep10')
 
-def send_results_to_api(data, result_url):
-    headers = {'Content-Type':'application/json'}
-    response = requests.post(result_url, json = data, headers=headers)
+# def send_results_to_api(data, result_url):
+#     headers = {'Content-Type':'application/json'}
+#     response = requests.post(result_url, json = data, headers=headers)
 
-    if response.status_code == 200:
-        return response.json
-    else:
-        return {'error':f"failed to send result to API: {response.status_code}"}
+#     if response.status_code == 200:
+#         return response.json
+#     else:
+#         return {'error':f"failed to send result to API: {response.status_code}"}
 
 def predict_text(params):
-    params = json.loads(params)
-
+    try:
+        params = json.loads(params)
+    except Exception as e:
+        logging.error(f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}")
+        return {"error": f"Invalid JSON input: {e.msg} at line {e.lineno} column {e.colno}"}
+        
     texts = params.get("texts",[])
-    api = params.get("api", "")
-    job_id = params.get("job_id","")
+    # api = params.get("api", "")
+    # job_id = params.get("job_id","")
+
+    if not texts:
+        return {"error": "Missing required parameters: 'urls'"}
 
     solutions = []
 
@@ -50,12 +57,12 @@ def predict_text(params):
         result = {'text':text, 'label':[label[pred_label]]}
         solutions.append(result)
 
-    result_url = f"{api}/{job_id}"
-    send_results_to_api(solutions, result_url)
-    return json.dumps({"solutions":solutions}, indent=4)
+    # result_url = f"{api}/{job_id}"
+    # send_results_to_api(solutions, result_url)
+    return json.dumps({"solutions":solutions})
 
 
-inputt = gr.Textbox(label="Parameters in Json Format...")
+inputt = gr.Textbox(label="Parameters in Json Format... Eg. {'texts':['text1', 'text2']")
 outputt = gr.JSON()
 
 application = gr.Interface(fn = predict_text, inputs = inputt, outputs = outputt, title='Multi Text Classification with API Integration..')