Update app.py

app.py

@@ -1,39 +1,63 @@
 import os
 import google.generativeai as genai
 import gradio as gr
-
+import time
 
 GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
 genai.configure(api_key=GOOGLE_API_KEY)
 model = genai.GenerativeModel('gemini-2.0-flash-exp')
 generation_config = genai.GenerationConfig(temperature=0)
 
+input_file = None
+prompt = prompt = "You are an expert in medical report analysis. Analyze the given medical report and generate a summarized report of the same."\
+         "All the medical tests in the given medical report must be specified in the summarized report. The response generated should be well-structured with health tips based on the report."
 
-def upload(file):
-    if not file:
-        return "**ERROR!!! FILE NOT UPLOADED**"
-
-    prompt = "You are an expert in medical report analysis. Analyze the given medical report and generate a summarized report of the same."\
-         "All the medical tests in the given medical report must be specified in the summarized report. The response generated should be well-structured with health tips based on the report."\
-
-    file = genai.upload_file(file)
+def greet(message, chatbot):
+    global input_file
+    if not input_file:
+        raise gr.Error(message="File not uploaded !!!", print_exception=False)
     
-    contents = [prompt, file]
+    chatbot.append({"role": "user", "content": message})
+    yield chatbot
+
+    history = chatbot+["-1"]
+
+    global prompt, input_file
+    response = model.generate_content([prompt, input_file]).text
+    assistant_message = ""
+    for char in response:
+        assistant_message += char
+        history[-1] = {"role": "assistant", "content": assistant_message}
+        yield history
+        time.sleep(0.1)
+
+with gr.Blocks() as demo:
+    def upload_file(file):
+        global input_file
+        input_file = genai.upload_file(file)
     
-    response = model.generate_content(contents)
-
-    file.delete()
+    def clear_file():
+        global input_file
+        input_file.delete()
+        
+        input_file = None
 
-    return response.text
-
-
-demo = gr.Interface(
-    fn = upload,
-    inputs = gr.File(file_types=['.pdf']),    # Input: File input
-    outputs = gr.Markdown(), # Output: Textbox to display the extracted text
-    title = "Medical Report Summarizer",
-    description = "Note: This summary is based on the provided medical report and does not constitute medical advice. Please consult with your healthcare provider for any questions or concerns regarding your health."
-)
+        return [],""
+    
+    with gr.Row():
+        with gr.Column(scale=5):
+            chatbot = gr.Chatbot(type="messages", render_markdown=True, height=550)
+            message = gr.Textbox(label="Enter your prompt")
+            submit = gr.Button("Submit")
+            clear = gr.ClearButton([message, chatbot])  
+            
+            submit.click(greet, [message, chatbot], chatbot)
+        
+        with gr.Column(scale=1):
+            file_upload = gr.File(label="Upload PDF File", file_types=['.pdf', '.txt'], )
+            
+            file_upload.upload(upload_file, file_upload)
+            file_upload.clear(clear_file, None, [chatbot, message])
 
 
 demo.launch()