Update app.py

app.py

@@ -84,8 +84,12 @@ for message in st.session_state["chat_history"]:
     r, t = message["role"], message["parts"][0]["text"]
     st.markdown(f"**{r.title()}:** {t}")
 
+# If there is a model response, clear the user input
+if st.session_state.chat_history and st.session_state.chat_history[-1]["role"] == "model":
+    st.session_state.user_input = ""
+
 # User input
-user_input = st.text_area("", height=5, key="user_input") 
+user_input = st.text_area("", value=st.session_state.user_input, height=5, key="user_input") 
 
 # File uploader
 uploaded_files = st.file_uploader("Upload images here or paste screenshots", type=["png", "jpg", "jpeg"], accept_multiple_files=True, key="uploaded_files")
@@ -128,7 +132,7 @@ if clear_button:
     st.session_state["chat_history"] = []
     # Update progress bar
     progress_bar.progress(1)
-    
+
 # Handle user input
 if user_input:
     # Add user input to chat history
@@ -142,20 +146,24 @@ if user_input:
     )
 
     # Generate model response
-    if model_name == "gemini-pro":
-        model = genai.GenerativeModel('gemini-pro')
-        response = model.generate_content(
-            contents=[user_input],
-            generation_config=generation_config
-        )
-    elif model_name == "gemini-pro-vision":
-        images = [Image.open(file).convert('RGB') for file in uploaded_files]
-        image_prompts = [{'mime_type': 'image/png', 'data': image.tobytes()} for image in images]
-        model = genai.GenerativeModel('gemini-pro-vision')
-        response = model.generate_content(
-            contents=[user_input] + image_prompts,
-            generation_config=generation_config
-        )
+    try:
+        if model_name == "gemini-pro":
+            model = genai.GenerativeModel('gemini-pro')
+            response = model.generate_content(
+                contents=[user_input],
+                generation_config=generation_config
+            )
+        elif model_name == "gemini-pro-vision":
+            images = [Image.open(file).convert('RGB') for file in uploaded_files]
+            image_prompts = [{'mime_type': 'image/png', 'data': image.tobytes()} for image in images]
+            model = genai.GenerativeModel('gemini-pro-vision')
+            response = model.generate_content(
+                contents=[user_input] + image_prompts,
+                generation_config=generation_config
+            )
+    except Exception as e:
+        st.write(f"An error occurred: {e}")
+        return
 
     # Add model response to chat history
     st.session_state["chat_history"].append({"role": "model", "parts": [{"text": response}]})
@@ -177,4 +185,3 @@ if user_input:
 
     # Clear user input
     st.session_state.user_input = ""
-