Updated app.py

app.py

@@ -0,0 +1,111 @@
+import gradio as gr
+import requests
+
+def send_request(url):
+    
+    payload = [
+            {"role": "user", "content": "Azərbaycan Neft-Qaz Sektorunda Ekoloji Risklər haqqında məlumat ver."}
+        ]
+
+    try:
+        response = requests.post(url, json=payload)
+        
+    except Exception as e:
+        
+        return {
+            "Error Message": f"Error occurred while sending request. Error message: {e}"
+        }
+        
+    try: 
+        
+        response_json = response.json()
+        
+    except Exception as e:
+        
+        return {
+            "Error Message": f"Error occurred while converting to json. Error message: {e}",
+            "Response": response.content
+        }
+    
+    if isinstance(response_json, dict):
+        
+        if "sources" not in response_json.keys() or "answer" not in response_json.keys():
+            
+            return {
+                "Error Message": "Required keys are not included in dictionary. `sources` and `answer` should be included in response keys.",
+                "Response": response_json
+            }
+            
+        if isinstance(response_json['sources'], list) and isinstance(response_json['answer'], str):
+            
+            for source in response_json['sources']:
+                
+                if isinstance(source, dict):
+                    
+                    if "pdf_name" not in source.keys() or "page_number" not in source.keys() or "content" not in source.keys():
+                        
+                        return {
+                            "Error Message": "Each source should have `pdf_name`, `page_number` and `content` as dictionary."
+                        }
+                        
+                    if isinstance(source['pdf_name'], str) and isinstance(source['page_number'], int) and isinstance(source['content'], str):
+                        
+                        continue
+                        
+                    else:
+                        
+                        return {
+                            "Error Message": "`pdf_name` should be of type str, `page_number` should be of type int, `content` should be of type str.",
+                            "Response": response_json
+                        }
+                    
+                else:
+                    
+                    return {
+                        "Error Message": "Sources in the `sources` should be type of dict.",
+                        "Response": response_json
+                    }
+        
+        else:
+            
+            return {
+                "Error Message": "`sources` should be type of list and `answer` should be type of str.",
+                "Response": response_json
+            }
+    
+    else:
+        
+        return {
+            "Error Message": f"Response should be dictionary with keys `sources` and `answer`.",
+            "Response": response_json
+        }
+    
+    return response_json
+
+with gr.Blocks() as demo:
+
+    output = gr.JSON(
+        label="Output"
+    )
+
+    input_box = gr.Textbox(
+        label="Input",
+        lines=1,
+        placeholder="Type your text here..."
+    )
+
+    send_btn = gr.Button("Send")
+
+    send_btn.click(
+        fn=send_request,
+        inputs=input_box,
+        outputs=output
+    )
+
+    input_box.submit(
+        fn=send_request,
+        inputs=input_box,
+        outputs=output
+    )
+
+demo.launch()