Add key

src/app.py

@@ -16,7 +16,7 @@ from modules.claim_processing import generate_claim_report_pdf
 load_dotenv()
 
 _FILE_PATH = Path(__file__).parents[1]
-
+API_KEY = os.getenv("FIREWORKS_API_KEY")
 
 class ClaimsAssistantApp:
     def __init__(self):
@@ -162,16 +162,6 @@ class ClaimsAssistantApp:
 
                     gr.Markdown("## ⚙️ Configuration")
 
-                    val = os.getenv("FIREWORKS_API_KEY", "")
-
-                    api_key = gr.Textbox(
-                        label="Fireworks AI API Key",
-                        type="password",
-                        placeholder="Enter your Fireworks AI API key",
-                        value=val,
-                        info="Required for AI processing",
-                    )
-
                     gr.Markdown("## 📋 Instructions")
                     gr.Markdown(
                         """
@@ -319,29 +309,22 @@ class ClaimsAssistantApp:
                             )
 
             # Event Handlers
-            def handle_damage_analysis(image, api_key):
+            def handle_damage_analysis(image):
                 if image is None:
                     return (
                         "❌ Please upload an image first",
                         gr.update(visible=False),
                     )
 
-                if not api_key.strip():
-                    return (
-                        "❌ Please enter your Fireworks AI API key first",
-                        gr.update(visible=False),
-                    )
-
                 try:
                     # Update status to show processing
                     yield (
                         "🔄 Analyzing damage... Please wait",
                         gr.update(visible=False),
                     )
-
                     image_dict = pil_to_base64_dict(image)
                     self.damage_analysis = analyze_damage_image(
-                        image=image_dict, api_key=api_key
+                        image=image_dict, api_key=API_KEY
                     )
 
                     yield (
@@ -362,11 +345,8 @@ class ClaimsAssistantApp:
                 with self.transcription_lock:
                     self.live_transcription = text
 
-            def initialize_transcription_service(api_key):
+            def initialize_transcription_service():
                 """Initialize transcription service when audio starts"""
-                if not api_key.strip():
-                    return False
-
                 # Always disconnect and cleanup existing service
                 if self.transcription_service:
                     try:
@@ -380,13 +360,13 @@ class ClaimsAssistantApp:
                     self.live_transcription = ""
                 
                 # Create fresh service
-                self.transcription_service = FireworksTranscription(api_key)
+                self.transcription_service = FireworksTranscription(API_KEY)
                 self.transcription_service.set_callback(live_transcription_callback)
                 
                 self.is_recording = True
                 return self.transcription_service._connect()
 
-            def process_audio_stream(audio_tuple, api_key):
+            def process_audio_stream(audio_tuple):
                 """Process incoming audio stream for live transcription"""
                 if not audio_tuple:
                     # Audio stopped - cleanup service but KEEP the transcription text
@@ -404,8 +384,8 @@ class ClaimsAssistantApp:
 
                 # Starting new recording - always initialize fresh
                 if not self.is_recording:
-                    if not initialize_transcription_service(api_key):
-                        return "❌ Failed to initialize transcription service. Check your API key."
+                    if not initialize_transcription_service():
+                        return "❌ Failed to initialize transcription service. Please try again later."
 
                 try:
                     sample_rate, audio_data = audio_tuple
@@ -458,7 +438,7 @@ class ClaimsAssistantApp:
                 with self.transcription_lock:
                     return self.live_transcription
 
-            def handle_incident_processing(api_key):
+            def handle_incident_processing():
                 """Process the recorded transcription into structured incident data with function calling"""
                 if not self.live_transcription.strip():
                     return (
@@ -473,13 +453,6 @@ class ClaimsAssistantApp:
                     self.transcription_service = None
                 self.is_recording = False
 
-                if not api_key.strip():
-                    return (
-                        "❌ Please enter your Fireworks AI API key first",
-                        gr.update(visible=False),
-                        gr.update(visible=False),
-                    )
-
                 try:
                     # Update status
                     yield (
@@ -490,7 +463,7 @@ class ClaimsAssistantApp:
 
                     # Use enhanced Fireworks processing with function calling
                     incident_analysis = process_transcript_description(
-                        transcript=self.live_transcription, api_key=api_key
+                        transcript=self.live_transcription, api_key=API_KEY
                     )
 
                     # Convert Pydantic model to dict for JSON display
@@ -524,7 +497,7 @@ class ClaimsAssistantApp:
                     )
                     return None
 
-            def handle_report_generation(api_key):
+            def handle_report_generation():
                 """Generate comprehensive claim report as PDF using AI"""
                 if not self.damage_analysis or not self.incident_data:
                     return (
@@ -535,15 +508,6 @@ class ClaimsAssistantApp:
                         gr.update(open=False),
                     )
 
-                if not api_key.strip():
-                    return (
-                        "❌ Please enter your Fireworks AI API key first",
-                        "<p style='text-align: center; color: gray;'>PDF report will appear here after generation</p>",
-                        gr.update(visible=False),
-                        gr.update(visible=False),
-                        gr.update(open=False),
-                    )
-
                 try:
                     # Show processing status
                     yield (
@@ -638,13 +602,11 @@ class ClaimsAssistantApp:
             # Wire up the events
             analyze_btn.click(
                 fn=handle_damage_analysis,
-                inputs=[image_input, api_key],
+                inputs=[image_input],
                 outputs=[damage_status, damage_results],
             )
 
-            # Don't automatically clear transcription when audio changes
-            # Users can use the clear button if needed
-            
+
             # Add clear transcription button handler
             def clear_transcription():
                 """Manually clear the transcription text"""
@@ -668,7 +630,7 @@ class ClaimsAssistantApp:
             # Handle streaming audio for live transcription
             audio_input.stream(
                 fn=process_audio_stream,
-                inputs=[audio_input, api_key],
+                inputs=[audio_input],
                 outputs=[transcription_display],
                 show_progress="hidden",
             )
@@ -676,13 +638,11 @@ class ClaimsAssistantApp:
             # Updated to include function calls display
             process_incident_btn.click(
                 fn=handle_incident_processing,
-                inputs=[api_key],
                 outputs=[incident_status, function_calls_display, incident_results],
             )
 
             generate_report_btn.click(
                 fn=handle_report_generation,
-                inputs=[api_key],
                 outputs=[
                     report_status,
                     pdf_viewer,