Update app.py

app.py

@@ -5,59 +5,66 @@ 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')
+
+system_instruction = "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."
+
 generation_config = genai.GenerationConfig(temperature=0)
+model = genai.GenerativeModel('gemini-2.0-flash-exp', 
+                              system_instruction=system_instruction,
+                              generation_config=generation_config)
+
+chat = model.start_chat()
 
 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."
+stop_generation = 0
 
-def greet(message, chatbot):
-    global prompt, input_file
-    
-    if not input_file:
-        raise gr.Error(message="File not uploaded !!!", print_exception=False)
+def stop():
+    global stop_generation
+    stop_generation = 1
+
+def clear():
+    for file in genai.list_files():
+        file.delete()
+
+    global input_file
+    input_file = None
+
+def add_message(history, message):
+    for x in message["files"]:
+            history.append({"role": "user", "content": {"path": x}})
+    if message["text"] is not None:
+        history.append({"role": "user", "content": message["text"]})
     
-    chatbot.append({"role": "user", "content": message})
-    yield chatbot
+    return history, gr.MultimodalTextbox(value=None, interactive=False, submit_btn=False, stop_btn=True)
+
+def bot(history, message):
+    global input_file,stop_generation
 
-    history = chatbot+["-1"]
+    if message["files"]:
+        input_file = genai.upload_file(message["files"][0])
+        response = chat.send_message([message["text"], input_file]).text
+    else:
+        response = chat.send_message(message["text"]).text
+    
+    history.append({"role": "assistant", "content": ""})
+    for character in response:
+        if stop_generation:
+            break
 
-    response = model.generate_content([prompt, input_file]).text
-    assistant_message = ""
-    for char in response:
-        assistant_message += char
-        history[-1] = {"role": "assistant", "content": assistant_message}
+        history[-1]["content"] += character
+        time.sleep(0.05)
         yield history
-        time.sleep(0.1)
 
 with gr.Blocks() as demo:
-    def upload_file(file):
-        global input_file
-        input_file = genai.upload_file(file)
-    
-    def clear_file():
-        global input_file
-        input_file.delete()
-        
-        input_file = None
+    chatbot = gr.Chatbot(bubble_full_width=False, type="messages",height=550)
+    chat_input = gr.MultimodalTextbox(interactive=True, placeholder="Enter message or upload file...", show_label=False)
 
-        return [],""
+    chat_msg = chat_input.submit (add_message, [chatbot, chat_input], [chatbot, chat_input])
+    bot_msg = chat_msg.then(bot, [chatbot, chat_input], chatbot)
+    bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True, submit_btn=True, stop_btn=False, value=None), None, chat_input)
     
-    with gr.Row():
-        with gr.Column(scale=5):
-            chatbot = gr.Chatbot(type="messages", render_markdown=True, height=500)
-            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])
-
+    chat_input.stop(stop)
+    chatbot.clear(clear)
 
 demo.launch()