Fine Tune Output!

app.py

@@ -14,7 +14,6 @@ import re
 from reportlab.lib import colors
 import random
 import streamlit.components.v1 as components
-from openai import OpenAI
 
 # ======================
 # CONFIGURATION SETTINGS
@@ -270,20 +269,6 @@ def initialize_groq_client():
     
     return Groq(api_key=api_key)
 
-def initialize_openai_client():
-    """Create and validate OpenAI API client"""
-    load_dotenv()
-    api_key = os.getenv("OPENAI_API_KEY")
-    
-    # Check if API key is in session state (from user input)
-    if 'openai_api_key' in st.session_state and st.session_state.openai_api_key:
-        api_key = st.session_state.openai_api_key
-    
-    if not api_key:
-        # Return None if no API key - we'll handle this in the UI
-        return None
-    
-    return OpenAI(api_key=api_key)
 
 def encode_logo(image_path):
     """Encode logo image to base64"""
@@ -594,81 +579,13 @@ Format each section with appropriate headings and use bullet points for lists. B
         st.error(f"Groq API error: {str(e)}")
         return None
 
-def generate_radiology_report_openai(uploaded_file, client):
-    """Generate AI-powered radiology analysis using OpenAI API"""
-    base64_image, img_format = process_image_data(uploaded_file)
-    
-    if not base64_image:
-        return None
-    
-    try:
-        with st.spinner("Analyzing image with OpenAI..."):
-            # Add progress bar for visual feedback
-            progress_bar = st.progress(0)
-            for i in range(100):
-                # Update progress bar
-                progress_bar.progress(i + 1)
-                import time
-                time.sleep(0.025)  # Simulate processing time
-                
-            # Updated prompt to request the detailed, structured format
-            response = client.chat.completions.create(
-                model="gpt-4-vision-preview",  # Use OpenAI's vision model
-                messages=[
-                    {
-                        "role": "user",
-                        "content": [
-                            {
-                                "type": "text",
-                                "text": (
-                                    "As a radiologist, analyze this medical image and provide a comprehensive report with the following structure:\n\n"
-                                    "1. DIAGNOSIS: Clear statement of primary diagnosis\n\n"
-                                    "2. ETIOLOGY: List possible causes of the identified condition\n\n"
-                                    "3. RISK FACTORS: Bullet-point list of risk factors associated with the condition\n\n"
-                                    "4. PATHOPHYSIOLOGY: Brief explanation of the disease mechanism\n\n"
-                                    "5. CLINICAL FEATURES: Overview of typical clinical presentation\n\n"
-                                    "6. SIGNS AND SYMPTOMS: Bullet-point list of common signs and symptoms\n\n"
-                                    "7. INVESTIGATIONS: Diagnostic tests typically used for confirmation\n\n"
-                                    "8. MANAGEMENT: Divided into three parts:\n"
-                                    "   - Initial Stabilization\n"
-                                    "   - Medical Management\n"
-                                    "   - Surgical Management (if applicable)\n\n"
-                                    "9. PROGNOSIS: Expected outcomes and factors affecting prognosis\n\n"
-                                    "Format each section with appropriate headings and use bullet points for lists. If you identify abnormal findings, make sure to highlight these clearly."
-                                )
-                            },
-                            {
-                                "type": "image_url",
-                                "image_url": {
-                                    "url": f"data:image/{img_format.lower()};base64,{base64_image}"
-                                }
-                            }
-                        ]
-                    }
-                ],
-                max_tokens=3000  # Increased token limit for more detailed response
-            )
-            return response.choices[0].message.content
-    except Exception as e:
-        st.error(f"OpenAI API error: {str(e)}")
-        return None
-
 def generate_radiology_report(uploaded_file, api_choice='groq'):
-    """Generate report using the selected API"""
-    if api_choice == 'groq':
-        client = initialize_groq_client()
-        if client:
-            return generate_radiology_report_groq(uploaded_file, client)
-        else:
-            st.error("Failed to initialize Groq client. Please check your API key.")
-            return None
-    else:  # OpenAI
-        client = initialize_openai_client()
-        if client:
-            return generate_radiology_report_openai(uploaded_file, client)
-        else:
-            st.error("OpenAI API key is required. Please provide your API key.")
-            return None
+    client = initialize_groq_client()
+    if client:
+        return generate_radiology_report_groq(uploaded_file, client)
+    else:
+        st.error("Failed to initialize Groq client. Please check your API key.")
+        return None
 
 # ======================
 # UI COMPONENTS
@@ -779,17 +696,7 @@ def render_sidebar():
             options=["Groq"],
             index=0,  # Default to Groq
         )
-        
-        # If OpenAI is selected, show API key input
-        # if api_provider == "OpenAI":
-        #     openai_key = st.text_input(
-        #         "OpenAI API Key", 
-        #         type="password",
-        #         help="Enter your OpenAI API key",
-        #         value=st.session_state.get('openai_api_key', '')
-        #     )
-        #     if openai_key:
-        #         st.session_state.openai_api_key = openai_key
+
         
         st.markdown('</div>', unsafe_allow_html=True)
         
@@ -830,7 +737,7 @@ def render_sidebar():
             if st.button("▶️ Initiate Analysis", use_container_width=True):
                 with st.spinner("Processing image..."):
                     # Use the selected API provider
-                    api_choice = 'groq' if api_provider == "Groq" else 'openai'
+                    api_choice = 'groq' if api_provider == "Groq" else None
                     report = generate_radiology_report(uploaded_file, api_choice)
                     
                     if report: