Update app.py

app.py

@@ -2,6 +2,7 @@ from huggingface_hub import InferenceClient
 import gradio as gr
 import random
 import prompts
+
 client = InferenceClient(
     "mistralai/Mixtral-8x7B-Instruct-v0.1"
 )
@@ -13,6 +14,7 @@ def format_prompt(message, history):
     prompt += f" {bot_response}</s> "
   prompt += f"[INST] {message} [/INST]"
   return prompt
+
 agents =[
     "WEB_DEV",
     "AI_SYSTEM_PROMPT",
@@ -26,12 +28,12 @@ agents =[
     "HUGGINGFACE_FILE_DEV",
 
 ]
+
 def generate(
-        prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,
+        prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, file=None
 ):
     seed = random.randint(1,1111111111111111)
 
-
     agent=prompts.WEB_DEV
     if agent_name == "WEB_DEV":
       agent = prompts.WEB_DEV_SYSTEM_PROMPT
@@ -69,6 +71,11 @@ def generate(
         seed=seed,
     )
 
+    # Process the uploaded file
+    if file:
+        file_content = file.read().decode("utf-8") 
+        prompt = f"Here's the file I uploaded: {file_content} \n{prompt}"
+
     formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
@@ -76,7 +83,9 @@ def generate(
     for response in stream:
         output += response.token.text
         yield output
-    return output
+
+    # This line should be indented at the same level as the 'for' loop
+    return output 
 
 
 additional_inputs=[
@@ -128,8 +137,6 @@ additional_inputs=[
         interactive=True,
         info="Penalize repeated tokens",
     ),
-
-
 ]
 
 examples=[