Update app.py

app.py

@@ -2,12 +2,9 @@ import streamlit as st
 import os
 import pickle
 import time
-import tempfile
-import os
-
-import requests
-
+import g4f
 
+st.set_page_config(page_title="MEDICAL ASSISTANT")
 
 st.markdown(
     """
@@ -18,21 +15,13 @@ st.markdown(
             font-weight: bold;
         }
     </style>
-    <div class="title">🧠 Phsycadalleics Chatbot</div>
-    """, 
+    <div class="title">🧠 MEDICAL ASSISTANT</div>
+    """,
     unsafe_allow_html=True
 )
-if 'chat_history' not in st.session_state:
-    st.session_state.chat_history = []  # Define chat_history in the session state
-
 # Load and Save Conversations
 conversations_file = "conversations.pkl"
 
-if not os.path.exists(conversations_file):
-    # Create the file
-    with open(conversations_file, 'w') as f:
-        pass  # The file is created with this line, no need to write anything to it
-
 
 @st.cache_data
 def load_conversations():
@@ -42,8 +31,9 @@ def load_conversations():
     except (FileNotFoundError, EOFError):
         return []
 
+
 def save_conversations(conversations):
-    temp_conversations_file = "temp_" + conversations_file
+    temp_conversations_file = conversations_file
     with open(temp_conversations_file, "wb") as f:
         pickle.dump(conversations, f)
     os.replace(temp_conversations_file, conversations_file)
@@ -69,7 +59,6 @@ def display_chats_sidebar():
             if col1.button('Start New Chat', key="new_chat"):
                 st.session_state.current_conversation = []
                 st.session_state.conversations.append(st.session_state.current_conversation)
-                st.session_state.chat_history = []
 
         with col2:
             if col2.button('Clear All Chats', key="clear_all"):
@@ -84,58 +73,44 @@ def display_chats_sidebar():
                 chat_title = truncate_string(chat_title_raw)
                 if st.sidebar.button(f"{chat_title}", key=f"chat_button_{idx}"):
                     st.session_state.current_conversation = st.session_state.conversations[idx]
-                    
-
-def dummy_response(prompt_input, chat_history):
-    api_url = 'http://b5c9-34-125-138-225.ngrok.io/api/chat'
-
-    data_payload = {
-        'query': prompt_input,
-        'chat_history': chat_history 
-    }
-
-    response = requests.post(api_url, json=data_payload)
 
-    if response.status_code == 200:
-        response_data = response.json()
-        result = response_data.get('response')
-        st.session_state.chat_history = response_data.get('chat_history')
-    else:
-        # Handle the case where the API does not return a successful response
-        result = f"An error occurred: {response.text}"
-    
-    return result, chat_history
-
-    
 
 def main_app():
     for message in st.session_state.current_conversation:
         with st.chat_message(message["role"]):
             st.write(message["content"])
 
+    def generate_response(prompt_input):
+        string_dialogue = "You are a helpful Medical Assistant. You will Only respond to Medical related Queries. Say Sorry to any other Type of Queries."
+        for dict_message in st.session_state.current_conversation:
+            string_dialogue += dict_message["role"].capitalize() + ": " + dict_message["content"] + "\\n\\n"
+
+        prompt = f"{string_dialogue}\n  {prompt_input} Assistant: "
+        response_generator = g4f.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": prompt}],
+            stream=True,
+        )
+        return response_generator
+
     if prompt := st.chat_input('Send a Message'):
         st.session_state.current_conversation.append({"role": "user", "content": prompt})
-        print(' PROMPT :', prompt)
         with st.chat_message("user"):
             st.write(prompt)
 
         with st.chat_message("assistant"):
             with st.spinner("Thinking..."):
-                
-
-                response, st.session_state.chat_history = dummy_response(prompt, st.session_state.chat_history) 
-                full_response = ''
+                response = generate_response(prompt)
                 placeholder = st.empty()
+                full_response = ''
                 for item in response:
                     full_response += item
                     time.sleep(0.003)
                     placeholder.markdown(full_response)
                 placeholder.markdown(full_response)
-                print(' RESPONSE :', full_response)
-
-                st.session_state.current_conversation.append({"role": "assistant", "content": response})
+                st.session_state.current_conversation.append({"role": "assistant", "content": full_response})
                 save_conversations(st.session_state.conversations)
-        
-        
+
+
 display_chats_sidebar()
 main_app()