Updated Q/A script

app.py

@@ -122,6 +122,7 @@ def process_and_summarize(audio_file, translate, model_size, do_summarize=True):
 
         
 
+
 @spaces.GPU(duration=60)
 def answer_question(context, question):
     logger.info("Starting Q&A process")
@@ -137,18 +138,24 @@ def answer_question(context, question):
         
         outputs = qa_pipeline(messages, max_new_tokens=256)
         
-        # Extract the answer from the generated text
-        full_response = outputs[0]["generated_text"]
-        
-        # Split the response to get only the assistant's reply
-        assistant_response = full_response.split("assistant:")[-1].strip() if "assistant:" in full_response else full_response
+        # Extract the answer from the output
+        if isinstance(outputs, list) and len(outputs) > 0 and isinstance(outputs[0], dict):
+            for message in outputs:
+                if message.get('role') == 'assistant':
+                    answer = message.get('content', '')
+                    break
+            else:
+                answer = "No answer found in the model's response."
+        else:
+            answer = str(outputs)  # Fallback to string representation of outputs
         
         logger.info("Q&A process completed successfully")
-        return assistant_response
+        return answer
     except Exception as e:
         logger.error(f"Q&A process failed: {str(e)}")
         logger.error(traceback.format_exc())
-        return f"Error occurred during Q&A process. Please try again. Error: \n{str(e)}"
+        return f"Error occurred during Q&A process. Please try again. Error: {str(e)}"
+
 
 # Main interface
 with gr.Blocks() as iface: