Create app.py

app.py

@@ -0,0 +1,74 @@
+import gradio as gr
+import numpy as np
+from PIL import Image
+
+def classify_fish(image):
+    return {"Baby": 0.1, "Small": 0.2, "Medium": 0.5, "Large": 0.2}
+
+def emit_sound(frequency: int):
+    return f"Emitting sound at {frequency} Hz"
+
+def fish_classification_ui(image):
+    result = classify_fish(image)
+    return result
+
+def sound_control_ui(frequency):
+    return emit_sound(frequency)
+
+def chat_bot(message):
+    message = message.lower()
+    qa_pairs = {
+        "status": "🧠 AI Agent is running fine. Monitoring in progress.",
+        "fish": "📊 We detect Baby, Small, Medium, and Large fish. Currently focusing on high-value species.",
+        "sound": "🔊 Emitting pulsed low-frequency sound to attract fish safely.",
+        "species": "🐟 High-demand species detected include Rohu, Katla, Murrel.",
+        "how many": "📈 Estimated fish count: Baby - 20, Small - 15, Medium - 10, Large - 5.",
+        "quality": "✅ Fish health and activity levels look optimal.",
+        "ocr": "🔍 OCR scanning dam gate status and water markings.",
+        "ner": "🧠 NER detects and classifies fish species from scientific names in camera labels.",
+        "camera": "🎥 Camera feed processed via CNN for size and movement analysis.",
+        "size": "📐 Fish size is auto-categorized as Baby, Small, Medium, or Large using our model.",
+        "dead fish": "⚠️ No dead fish detected near the thrust gates currently.",
+        "thrust gate": "🚪 Monitoring open/close cycle and alerting when fish are near during thrust events.",
+        "maintenance": "🛠️ Regular AI system checks scheduled weekly.",
+        "data": "📊 Sensor data streamed every 2 seconds from riverbanks and gates.",
+        "alert": "🚨 Auto-alerts sent to fishermen if high movement or danger is detected.",
+        "chat": "💬 I can assist engineers and fishermen with system status, fish count, and more.",
+        "retrain": "🧑‍💻 Retraining is supported manually with new images via UI.",
+        "dashboard": "📊 The live dashboard shows fish count, sound activity, and gate logs.",
+        "developer": "👨‍💻 Developed by EchoFishAI with Gradio, CNN, and RL techniques.",
+        "help": "📘 You can ask about fish, gate status, sound, camera, or system alerts."
+    }
+
+    for key in qa_pairs:
+        if key in message:
+            return qa_pairs[key]
+    return "🤖 Hello! Ask me about fish, sound, species, status, gates, or AI system help."
+
+with gr.Blocks() as demo:
+    gr.Markdown("# EchoFishAI 🐟")
+    gr.Markdown("Smart Fish Tracking & Attraction System with AI")
+
+    with gr.Tab("Fish Classifier"):
+        with gr.Row():
+            image_input = gr.Image(type="pil")
+            output = gr.Label()
+        classify_btn = gr.Button("Classify Fish")
+        classify_btn.click(fish_classification_ui, inputs=image_input, outputs=output)
+
+    with gr.Tab("Sound Emission"):
+        with gr.Row():
+            freq_input = gr.Slider(minimum=10, maximum=1000, step=10, label="Frequency (Hz)")
+            sound_output = gr.Textbox()
+        emit_btn = gr.Button("Emit Sound")
+        emit_btn.click(sound_control_ui, inputs=freq_input, outputs=sound_output)
+
+    with gr.Tab("Ask AI Agent"):
+        with gr.Row():
+            chatbot_input = gr.Textbox(label="Ask something...")
+            chatbot_output = gr.Textbox(label="Response")
+        chat_btn = gr.Button("Send")
+        chat_btn.click(chat_bot, inputs=chatbot_input, outputs=chatbot_output)
+
+if __name__ == "__main__":
+    demo.launch(debug=True)