updated app.py

fixed all errors

app.py

@@ -20,31 +20,36 @@ def vision(img):
     vision_output = response.text
     return vision_output
 
+
+        
 def chat(inpt_frm_vsn):
     model = genai.GenerativeModel('gemini-pro')
     chat = model.start_chat(history=[])
 
     message_count = 0
 
-    res = inpt_frm_vsn + 'Tell me the fruit name and nutrition'
+    continue_chat = True
+
+    while continue_chat:
+        if message_count == 0:
+            input_msg = inpt_frm_vsn + ' Tell me the fruit name and nutrition. Complete answer in 55 words. Never show the word count.'
+        else:
+            input_key = f"chat_input_{message_count}"
+            user_input = st.chat_input('Please write your message here...', key=input_key)
+            if user_input:
+                input_msg = user_input + ' Complete answer in 55 words. Never show the word count.'
+            else:
+                continue_chat = False
+                break
 
-    while True:
-        con = '. Complete answer in 55 word. Never show the word count.'
-        input_msg = res + con
         response = chat.send_message(input_msg, generation_config=genai.types.GenerationConfig(
             max_output_tokens=100
         ))
         to_markdown(response.text)
-
         st.text('Gemini: ' + response.text)
-        
-        input_key = f"chat_input_{message_count}"
+        message_count += 1
 
-        res = st.chat_input('Please write your message here...', key=input_key)
-        if res:
-            st.write(f"User: {res}")
         
-        message_count += 1
 
 def main():
     st.title("Gemini")