Update app.py

app.py

@@ -171,33 +171,28 @@ def handle_no_answer(response):
 
 def ask_bot(query):
     # Definiere den standardmäßigen Prompt
-    standard_prompt = ("Schreibe immer höflich und auf antworte immer in der Sprache, "
-                       "in der der User auch schreibt. Formuliere immer ganze freundliche "
-                       "ganze Sätze und biete, wenn möglich, auch mehr Informationen "
-                       "(aber nicht mehr als 1 Satz mehr). Wenn der User sehr vage schreibt, "
-                       "frage nach - gerade wenn es um Jahre geht, frage nur nach Jahren, "
-                       "wenn du zu dem Thema auch Daten zu bestimmten Jahren hast.")
+    standard_prompt = "Schreibe immer höflich und auf antworte immer in der Sprache in der der User auch schreibt. Formuliere immer ganze freundliche ganze Sätze und biete wenn möglich auch mehr Informationen (aber nicht mehr als 1 Satz mehr). Wenn der User sehr vage schreibt frage nach. Wenn du zu einer bestimmten Frage Daten aus mehreren Jahren hast, frage den User für welche Jahre er sich interessiert und nenne ihm natürlich Möglichkeiten über die Jahre die du hast. "
     # Kombiniere den standardmäßigen Prompt mit der Benutzeranfrage
-    full_query = standard_prompt + " " + query
+    full_query = standard_prompt + query
     return full_query
 
-
-# Funktion zum sicheren Hinzufügen zur Chat-Historie
 def add_to_chat_history(speaker, message, history_key):
     if history_key in st.session_state:
-        st.session_state[history_key].append((speaker, message))
-
+        st.session_state[history_key].append({"speaker": speaker, "message": message})
 
+def construct_context(chat_history):
+    # Limit the context to the last few interactions (e.g., last 5)
+    limited_history = chat_history[-5:]
+    return " ".join([f"{entry['speaker']}: {entry['message']}" for entry in limited_history])
 
-# Funktion zum sicheren Kombinieren der Chat-Historie mit der aktuellen Anfrage
-def combine_history_with_query(current_query, history):
-    combined_history = ""
-    for chat in history:
-        # Stelle sicher, dass das Tupel zwei Elemente hat
-        if len(chat) >= 2:
-            combined_history += f"{chat[0]}: {chat[1]} "
-    return combined_history + current_query
+def display_latest_messages(chat_history):
+    latest_messages = chat_history[-2:]
+    for chat_entry in latest_messages:
+        speaker, message = chat_entry['speaker'], chat_entry['message']
+        background_color = "#ffeecf" if speaker == "User" else "#e0ecff"
+        st.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{speaker}: {message}</div>", unsafe_allow_html=True)
 
+    
 
 def page1():
     try:
@@ -261,53 +256,50 @@ def page1():
                 if st.button("Was bedeutet die Vorhaltefinanzierung?"):
                     query = "Was bedeutet die Vorhaltefinanzierung?"
 
+
+        
             if query:
-                st.text(f"Type of query: {type(query)}")
-                standard_prompt = ("Schreibe immer höflich und antworte immer in der Sprache, "
-                                   "in der der User auch schreibt. Formuliere immer ganze freundliche "
-                                   "Sätze und biete, wenn möglich, auch mehr Informationen "
-                                   "(aber nicht mehr als 1 Satz mehr). Wenn der User sehr vage schreibt, "
-                                   "frage nach - gerade wenn es um Jahre geht, frage nur nach Jahren, "
-                                   "wenn du zu dem Thema auch Daten zu bestimmten Jahren hast.")
-                full_query = standard_prompt + " " + query
-                combined_query = combine_history_with_query(full_query, st.session_state['chat_history_page1'])
-    
+                full_query = ask_bot(query)  # Generate full query using ask_bot function
+            
+                # Start timing
+                start_time = time.time()
+            
                 with st.spinner('Bot is thinking...'):
+                    # Load the chatbot model
                     chain = load_chatbot()
-                    docs = VectorStore.similarity_search(query=combined_query, k=5)
+            
+                    # Perform the similarity search and get the response
+                    docs = VectorStore.similarity_search(query=full_query, k=5)
                     with get_openai_callback() as cb:
-                        response = chain.run(input_documents=docs, question=combined_query)
+                        response = chain.run(input_documents=docs, question=full_query)
                         response = handle_no_answer(response)
+            
+                # Stop timing
+                end_time = time.time()
+            
+                # Calculate duration
+                duration = end_time - start_time
+                st.text(f"Response time: {duration:.2f} seconds")
+            
+                # Update chat history after the bot's response
                 st.session_state['chat_history_page1'].append(("User", query))
                 st.session_state['chat_history_page1'].append(("Bot", response))
-    
-                # Diagnostic print: Display the entire chat history
-                st.text(f"Current chat history: {st.session_state['chat_history_page1']}")
-
+            
                 # Display new messages at the bottom
                 new_messages = st.session_state['chat_history_page1'][-2:]
                 for chat in new_messages:
-                    try:
-                        # Ensure the tuple has at least two elements
-                        if len(chat) < 2:
-                            raise ValueError(f"Invalid chat tuple: {chat}")
-                
-                        speaker, message = chat[:2]
-                        # No need to extract the status as it is no longer used
-                        background_color = "#ffeecf" if speaker == "User" else "#ffeecf"
-                        new_messages_placeholder.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{speaker}: {message}</div>", unsafe_allow_html=True)
-                    except IndexError as e:
-                        st.error(f"Error processing chat entry: {e}")
-                    except Exception as e:
-                        st.error(f"Unexpected error: {e}")
-                    
+                    background_color = "#ffeecf" if chat[2] == "new" else "#ffeecf" if chat[0] == "User" else "#ffeecf"
+                    new_messages_placeholder.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{chat[0]}: {chat[1]}</div>", unsafe_allow_html=True)
+            
+                # Clear the input field after the query is made
                 query = ""
+                
+# Mark all messages as old after displaying
+st.session_state['chat_history_page1'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history_page1']]
 
-                st.session_state['chat_history_page1'] = [(sender, msg) for sender, msg in st.session_state['chat_history_page1']]
-            
     except Exception as e:
         st.error(f"Upsi, an unexpected error occurred: {e}")
-
+        # Optionally log the exception details to a file or error tracking service