Update app.py

app.py

@@ -11,6 +11,7 @@ client = None
 def initialize_client(api_key=None):
     global client
     api_key = os.getenv("TOGETHER_API_KEY")
+    print(api_key)
     client = Together()
 
 def encode_image(image_path):
@@ -32,7 +33,7 @@ def bot_streaming(image_path, history):
 
     prompt = """
       You are expert in computer vision and autos inspection from image analysis. Inspect the auto's radiator core support for damage resulting from a front-end
-      collision. Check for any bent, cracked, or misaligned components, including the core support bracket, mounting points,
+      collision. Check for any bent,  cracked, or misaligned components, including the core support bracket, mounting points,
       and frame sections.
       Ensure that the radiator is securely mounted and assess if there are any signs of stress or deformation caused by the accident. 
       then tell it as accidental if not then not accidental.
@@ -82,61 +83,17 @@ def bot_streaming(image_path, history):
         history[-1] = ("Image uploaded", error_message)
         yield history
 
-# Define data for the Body frame accident checklist
-body_frame_data = [
-    {"label": "Radiator Core Support"},
-    {"label": "Right Strut Tower Apron"},
-    {"label": "Left Strut Tower Apron"},
-    {"label": "Right Front Rail"},
-    {"label": "Left Front Rail"},
-    {"label": "Cowl Panel Firewall"},
-    {"label": "Right A Pillar"},
-    {"label": "Left A Pillar"},
-    {"label": "Right B Pillar"},
-    {"label": "Left B Pillar"},
-    {"label": "Right C Pillar"},
-    {"label": "Left C Pillar"},
-    {"label": "Boot Floor"},
-    {"label": "Boot Lock Pillar"},
-    {"label": "Rear Sub Frame"},
-    {"label": "Front Sub Frame"},
-]
-
-# Define data for the Tyres
-tyre_data = [
-    {"label": "Front Right Tyre"},
-    {"label": "Front Left Tyre"},
-    {"label": "Rear Right Tyre"},
-    {"label": "Rear Left Tyre"},
-    {"label": "Tyre Size"},
-    {"label": "Rims"},
-    {"label": "Wheel Caps"},
-]
-
-def create_category_section(data, name):
-    with gr.Row():
-        gr.Markdown(f"## {name}")
-        for item in data:
-            with gr.Column():
-                gr.Markdown(f"### {item['label']}")
-                image = gr.Image(type="filepath", label="Upload Image")
-                chatbot = gr.Chatbot()
-                image.upload(bot_streaming, inputs=[image, chatbot], outputs=chatbot)
-
-# Create the main interface
+# Set up Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Car Radiator and Parts Accident Detection")
-    gr.Markdown("Upload an image of different parts to determine if it shows signs of an accident")
+    gr.Markdown("# Radiator Accident Detection")
+    gr.Markdown("Upload an image of a radiator to determine if it shows signs of an accident")
 
-    # Display Body Frame Section
-    create_category_section(body_frame_data, "Body Frame Accident Checklist")
-    
-    # Display Tyres Section
-    create_category_section(tyre_data, "Tyres")
+    chatbot = gr.Chatbot()
+    img = gr.Image(type="filepath", label="Upload Radiator Image")
+    clear = gr.Button("Clear")
 
-    # Clear button to reset chatbot history
-    clear = gr.Button("Clear All")
-    clear.click(lambda: None, None, None)
+    img.upload(bot_streaming, inputs=[img, chatbot], outputs=chatbot)
+    clear.click(lambda: None, None, chatbot, queue=False)
 
 if __name__ == "__main__":
     demo.launch(debug=True)