Update app.py

app.py

@@ -87,7 +87,7 @@ def respond(
     programming_language,
     *args,
     **kwargs,
-) -> str:
+) -> dict:
     # Initialize the model
     model = pipeline("text-generation", model="enricoros/big-agi")
 
@@ -103,7 +103,11 @@ def respond(
     # Extract the assistant's response
     assistant_response = response[0]['generated_text'].strip()
 
-    return assistant_response
+    return {
+        'assistant_response': assistant_response,
+        'severity': severity,
+        'programming_language': programming_language,
+    }
         )
 
 with gr.Blocks() as demo: