Update app.py

app.py

@@ -158,7 +158,7 @@ def normalize_text(text):
     return ' '.join(words)
 
 llm = HuggingFaceEndpoint(
-    repo_id="mistralai/Mistral-7B-Instruct-v0.2",
+    repo_id="HuggingFaceH4/starchat2-15b-v0.1",
     task="text-generation",
     max_new_tokens=4096,
     temperature=0.6,
@@ -194,8 +194,9 @@ def user3(user_message, history, file_path):
     return "", history + [[combined_message, None]], file_path
 
 messages1 = [
-    HumanMessage(content="You are a helpful assistant."),
-    AIMessage(content="Sure, how can I assist you today?")]
+  SystemMessage(content="You are a helpful assistant."),
+  HumanMessage(content="Hi AI, how are you today?"),
+AIMessage(content="I'm great thank you. How can I help you?")]
 messages2 = messages1.copy()
 messages3 = messages1.copy()
 messages4 = messages1.copy()
@@ -213,10 +214,10 @@ def Chat_Message(history):
       messages1 = messages1[-8:]
 
     history[-1][1] = ""
-
-    history[-1][1] += response.content
-
-    return history
+    for character in response.content:
+        history[-1][1] += character
+        time.sleep(0.0025)
+        yield history
 
 def Web_Search(history):
     global messages2
@@ -235,7 +236,7 @@ def Web_Search(history):
     Query: {message}
     Search Content:
     {source_knowledge}
-
+    
     #If the query is not related to specific Search Content, engage in general conversation or provide relevant information from other sources.
     """
 
@@ -248,10 +249,10 @@ def Web_Search(history):
       messages2 = messages2[-8:]
 
     history[-1][1] = ""
-
-    history[-1][1] += response.content
-
-    return history
+    for character in response.content:
+        history[-1][1] += character
+        time.sleep(0.0025)
+        yield history
 
 def Chart_Generator(history):
     global messages3
@@ -292,10 +293,10 @@ def Chart_Generator(history):
         combined_content=response.content
         
     history[-1][1] = ""
-
-    history[-1][1] += combined_content
-
-    return history
+    for character in combined_content:
+        history[-1][1] += character
+        time.sleep(0.0025)
+        yield history
 
 def Link_Scratch(history):
     global messages4
@@ -326,7 +327,7 @@ def Link_Scratch(history):
             Query: {user_message}
             Link Content:
             {content_data}
-
+            
             #If the query is not related to specific Link Content, engage in general conversation or provide relevant information from other sources.
             """
             message = HumanMessage(content=augmented_prompt)
@@ -340,10 +341,10 @@ def Link_Scratch(history):
             response_message = response.content
 
     history[-1][1] = ""
-
-    history[-1][1] += response_message
-
-    return history
+    for character in response_message:
+        history[-1][1] += character
+        time.sleep(0.0025)
+        yield history
 
 def insert_line_breaks(text, every=8):
     return '\n'.join(text[i:i+every] for i in range(0, len(text), every))
@@ -384,7 +385,7 @@ def File_Interact(history,filepath):
             Query: {user_message}
             File Content:
             {content_data}
-
+            
             #If the query is not related to specific File Content, engage in general conversation or provide relevant information from other sources.
             """
             message = HumanMessage(content=augmented_prompt)
@@ -398,10 +399,10 @@ def File_Interact(history,filepath):
             response_message = response.content
 
     history[-1][1] = ""
-
-    history[-1][1] += response_message
-
-    return history
+    for character in response_message:
+        history[-1][1] += character
+        time.sleep(0.0025)
+        yield history
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
   with gr.Row():