Update backend/main.py

backend/main.py

@@ -1,43 +1,35 @@
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
-import random
+import os
+
+# Import your simulation and AI scripts
 from backend.comm_system import simulate_channel
 from backend.ai_decoder import decode_semantic_intent
 
 app = FastAPI()
 
-class ChannelRequest(BaseModel):
-    snr_db: float
-    message: str
-
-@app.post("/api/simulate")
-def simulate_transmission(req: ChannelRequest):
-    # 1. Run the physics simulation for the UI diagram
-    constellation_data = simulate_channel(req.snr_db)
-    
-    # 2. Simulate traditional Bit Error Rate (BER) text corruption
-    # Lower SNR = More errors in the string
-    corrupted = list(req.message)
-    error_rate = max(0, (15 - req.snr_db) / 30) 
-    
-    for i in range(len(corrupted)):
-        if random.random() < error_rate and corrupted[i] != " ":
-            corrupted[i] = random.choice("!@#$%^&*~?")
-            
-    corrupted_message = "".join(corrupted)
-    
-    # 3. Trigger Generative AI to fix the signal if corruption occurred
-    if error_rate > 0:
-        recovered_message = decode_semantic_intent(corrupted_message)
-    else:
-        recovered_message = req.message
+# This tells Python exactly what the React app is sending
+class MessageRequest(BaseModel):
+    text: str
 
-    return {
-        "constellation": constellation_data,
-        "corrupted_message": corrupted_message,
-        "recovered_message": recovered_message
-    }
+# 📡 The endpoint that listens for the "Transmit Message" button
+@app.post("/api/transmit")
+async def transmit_message(req: MessageRequest):
+    try:
+        # 1. Pass the user's text through the noisy channel
+        corrupted_text = simulate_channel(req.text)
+        
+        # 2. Ask the Hugging Face AI to fix it
+        decoded_text = decode_semantic_intent(corrupted_text)
+        
+        # 3. Send both back to the React UI
+        return {
+            "corrupted": corrupted_text,
+            "decoded": decoded_text
+        }
+    except Exception as e:
+        return {"corrupted": "Error in channel", "decoded": f"Backend Error: {str(e)}"}
 
-# Mount the React App (Must be at the bottom so it doesn't block /api)
+# Mount the React frontend so the browser can see it
 app.mount("/", StaticFiles(directory="frontend/dist", html=True), name="static")