Update app.py

app.py

@@ -7,7 +7,7 @@ from mistralai.models.chat_completion import ChatMessage
 client = MistralClient(api_key= os.getenv('mistral_api_key'))
 
 # Function to get chat response
-def get_mistral_response(user_input, history: list[tuple[str, str]], context):
+def get_mistral_response(user_input: str, history: list[tuple[str, str]], context: str):
     messages = [ChatMessage(role='user', content='you are helpful assistant answering question based on given context. provide a highly consie and precise answer, along with a citation from original context in the format of JSON.'),
                 ChatMessage(role='assistant', content='I understand.')] # system prompt has been baked into fine-tuned model. 
     
@@ -29,7 +29,7 @@ def get_mistral_response(user_input, history: list[tuple[str, str]], context):
 demo = gr.ChatInterface(
     get_mistral_response,
     title='no-nonsense QA bot',
-    description="After fine-tuning, the bot answers your question with grounded citation. Copy paste your contextual information in the box below.",
+    description="After fine-tuning, the bot answers your question with a grounded citation. Paste your contextual information in the box below.",
     additional_inputs=[
         gr.Textbox(value="", label="Answer will be based on the context input here", lines=10),
     ],