Upload app.py

app.py

@@ -513,7 +513,7 @@ def user_input(user_question):
         {"input_documents":docs, "question": user_question}
         , return_only_outputs=True)
         
-    pdfof = pdfans["output_text"]
+    pdfof = f'{pdfans["output_text"]}'
     
     return pdfof
     
@@ -782,14 +782,25 @@ if pdf_mode:
 if pdf_mode:
     prompt = None
     prompty = st.chat_input("Write your questions according to the pdf")
-    if prompty:      
-        prmt  = {'role': 'user', 'parts':[prompty]}
-        usertxt = user_input(prompty)
-        append_message({'role': 'model', 'parts':usertxt})
 else:
     prompty = None
     prompt = st.chat_input("Write your message")
 
+if prompty:
+    spinertxt = 'Wait a moment, I am thinking...'
+    prmt  = {'role': 'user', 'parts':[prompty]}         
+    append_message(prmt)   
+    with st.spinner(spinertxt):
+        if len(prmt['parts'][0]) > 1:
+            usertxt = user_input(prompty)
+                  
+        try:
+            append_message({'role': 'model', 'parts':usertxt})
+        except Exception as e:
+            append_message({'role': 'model', 'parts':f'{type(e).__name__}: {e}'})
+                
+        st.rerun()
+        
 if prompt:
     txt = ''