Update app.py

app.py

@@ -31,7 +31,7 @@ def load_qa_model():
         logger.info(f"Q&A model loaded successfully")
         return qa_pipeline
     except Exception as e:
-        logger.warning(f"Failed to load Q&A model. Error: {str(e)}")
+        logger.warning(f"Failed to load Q&A model. Error: \n{str(e)}")
         return None
 
 
@@ -43,7 +43,7 @@ def load_summarization_model():
         logger.info(f"Summarization model loaded successfully on {'GPU' if cuda_available else 'CPU'}")
         return summarizer
     except Exception as e:
-        logger.warning(f"Failed to load summarization model on GPU. Falling back to CPU. Error: {str(e)}")
+        logger.warning(f"Failed to load summarization model on GPU. Falling back to CPU. Error: \n{str(e)}")
         summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", device=-1)
         logger.info("Summarization model loaded successfully on CPU")
         return summarizer
@@ -121,6 +121,7 @@ def process_and_summarize(audio_file, translate, model_size, do_summarize=True):
         raise gr.Error(f"Processing failed: {str(e)}")
 
         
+
 @spaces.GPU(duration=60)
 def answer_question(context, question):
     logger.info("Starting Q&A process")
@@ -135,18 +136,19 @@ def answer_question(context, question):
         ]
         
         outputs = qa_pipeline(messages, max_new_tokens=256)
-        answer = outputs[0]["generated_text"]
         
         # Extract the answer from the generated text
-        answer = answer.split("assistant:")[-1].strip()
+        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
         
         logger.info("Q&A process completed successfully")
-        return answer
+        return assistant_response
     except Exception as e:
         logger.error(f"Q&A process failed: {str(e)}")
         logger.error(traceback.format_exc())
-        return "Error occurred during Q&A process. Please try again."
-
+        return f"Error occurred during Q&A process. Please try again. Error: \n{str(e)}"
 
 # Main interface
 with gr.Blocks() as iface: