fixed the chat reset

app.py

@@ -16,12 +16,16 @@ if "messages" not in st.session_state:
 if "model" not in st.session_state:
     st.session_state.model = genai.GenerativeModel(MODEL_ID)
 
-model = st.session_state.model
+if "chat" not in st.session_state:
+    st.session_state.chat = st.session_state.model.start_chat()
+
+if "is_new_file" not in st.session_state:
+    st.session_state.is_new_file = True
 
 # Function to reset chat history
 def reset_chat():
     st.session_state.messages = []
-    model.start_chat() 
+    st.session_state.model.start_chat() 
 
 def main():
     # Streamlit app
@@ -45,51 +49,70 @@ def main():
         else:
             st.error("Unsupported file type. Please upload an image or PDF.")
             st.stop()
-
+        
         # Reset chat history when a new file is uploaded
         reset_chat()
-
-        # Text input for user prompt
-        user_input = st.text_area("Enter your prompt:", height=200)
-
-        # Send button
-        if st.button("Send"):
-            if user_input:
-                # Add user message to chat history
-                st.session_state.messages.append({"role": "user", "content": user_input})
-
-            # Display chat history
-            for message in st.session_state.messages:
-                with st.chat_message(message["role"]):
-                    st.markdown(message["content"])
-
-            with st.spinner("Processing..."):
+        st.session_state.is_new_file = True
+
+    # Text input for user prompt
+    user_input = st.text_area("Enter your prompt:", height=200)
+
+    # Send button
+    if st.button("Send"):
+        if not uploaded_file or not user_input:
+            st.warning("Please upload an image or PDF and enter a prompt.")
+            st.stop()   
+        
+        if user_input:
+            # Add user message to chat history
+            st.session_state.messages.append({"role": "user", "content": user_input})
+
+        with st.spinner("Processing..."):
+            if st.session_state.is_new_file:
                 # Upload the file with the correct MIME type
                 file_data = genai.upload_file(uploaded_file, mime_type=mime_type)
 
-                # Send file and prompt to Gemini API with streaming enabled
-                response = model.generate_content(
+                # Send file and prompt to Gemini API
+                chat = st.session_state.chat
+                response = chat.send_message(
                     [
                         user_input, 
                         file_data
                     ],
                     stream=enable_stream  
                 )
+                st.session_state.is_new_file = False
+            else:
+                # continue chat without sending the file again
+                # Send a text prompt to Gemini API
+                chat = st.session_state.chat
+                response = chat.send_message(
+                    [
+                        user_input
+                    ],
+                    stream=enable_stream  
+                )
 
-                # Display Gemini response as it streams in
-                full_response = ""
-                if enable_stream:
-                    for chunk in response:
-                        with st.chat_message("assistant"):
-                            st.write(chunk.text)
-                        full_response += chunk.text 
-                else:                
-                    full_response = response.text
+            # Display Gemini response as it streams in
+            full_response = ""
+            if enable_stream:
+                for chunk in response:
                     with st.chat_message("assistant"):
-                        st.write(full_response)                    
+                        st.write(chunk.text)
+                    full_response += chunk.text 
+            else:                
+                full_response = response.text
+                with st.chat_message("assistant"):
+                    st.write(full_response)                    
+
+            # Add Gemini response to chat history
+            st.session_state.messages.append({"role": "assistant", "content": full_response})
 
-                # Add Gemini response to chat history
-                st.session_state.messages.append({"role": "assistant", "content": full_response})
+            st.subheader("Chat History")
+            # Display chat history
+            for message in st.session_state.messages:
+                with st.chat_message(message["role"]):
+                    st.markdown(message["content"])
 
 if __name__ == "__main__":
     main()