remove thinking

app.py

@@ -2,21 +2,19 @@ import gradio as gr
 import fitz  
 import os
 import easyocr  
+import re  
 from huggingface_hub import InferenceClient
 
 HF_API_KEY = os.getenv("HF_API_KEY")
 
-# Initialize the Hugging Face Inference Client
 client = InferenceClient(
-    provider="together",  # Specify provider
+    provider="together",  
     api_key=HF_API_KEY
 )
 
-# Initialize EasyOCR
 reader = easyocr.Reader(['en'])
 
 def extract_text_from_pdf(pdf_file):
-    """Extracts text from the uploaded PDF file."""
     text = ""
     with fitz.open(pdf_file.name) as doc:
         for page in doc:
@@ -24,12 +22,10 @@ def extract_text_from_pdf(pdf_file):
     return text.strip()
 
 def extract_text_from_image(image_file):
-    """Extracts text from an uploaded image file using EasyOCR."""
     text = reader.readtext(image_file.name, detail=0)
     return "\n".join(text).strip()
 
 def analyze_medical_report(file):
-    """Determines file type (PDF or Image) and extracts text accordingly."""
     if file.name.lower().endswith(".pdf"):
         text = extract_text_from_pdf(file)
     else:
@@ -38,35 +34,41 @@ def analyze_medical_report(file):
     if not text:
         return "No text found in the uploaded document."
 
-    # Construct message for DeepSeek AI
     messages = [
         {"role": "user", "content": f"""
-        Analyze the following medical report and provide a structured response:
-        
-        1. **Short Description**: Briefly summarize the report in 2-3 sentences.
-        2. **Key Concerns**: List any health issues found.
-        3. **Recommendations**: Provide necessary medical advice or next steps.
-        
-        Report:
-        {text}
-        """}
+    Analyze the following medical report and provide a structured response in the exact format below. 
+    Ensure the response follows this structure, dont use extra space
+    **Short Description:**  
+    [Briefly summarize the report in 2-3 sentences. Include key test details.]
+    **Key Concerns:**
+    explain in Ordered form. First tell the test name value and tell is it high or low. Keep it short
+    2-3 main concerns are enough
+    1. **[Test Name (Value)]:** [High or low Explain what the abnormality suggests.]  
+
+    **Recommendations:**  
+    Give in bullet points. Give only 2-3 and dont write too much explanation
+
+    Medical Report:  
+    {text}
+    """}
     ]
 
     try:
-        # Send request to Hugging Face Inference API
         completion = client.chat.completions.create(
             model="deepseek-ai/DeepSeek-R1", 
             messages=messages, 
             max_tokens=500,
         )
 
-        # Extract response
-        return completion.choices[0].message.content if completion.choices else "No response generated."
+        output = completion.choices[0].message.content if completion.choices else "No response generated."
+        
+        output = re.sub(r"<think>.*?</think>", "", output, flags=re.DOTALL).strip()
+        
+        return output
     
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Gradio Interface
 interface = gr.Interface(
     fn=analyze_medical_report,
     inputs=gr.File(type="filepath", label="Upload Medical Report (PDF/Image)"),