Update app.py

app.py

@@ -47,11 +47,14 @@ devices = {
 
 # Function to Generate Code
 def generate_code(prompt, model="llama-3.3-70b-versatile"):
-    chat_completion = client.chat.completions.create(
-        messages=[{"role": "user", "content": prompt}],
-        model=model,
-    )
-    return chat_completion.choices[0].message.content
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[{"role": "user", "content": prompt}],
+            model=model,
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        return f"Error during code generation: {str(e)}"
 
 # Function to Chunk and Tokenize Code
 def tokenize_and_chunk_code(code, model_name="gpt2", max_length=512):
@@ -82,6 +85,10 @@ def code_generator(device, prompt):
     # Generate Code
     generated_code = generate_code(f"Write {device} code for: {prompt}")
 
+    # Handle if code generation fails
+    if "Error" in generated_code:
+        return generated_code  # Return the error message if the generation fails
+
     # Chunk, Tokenize, and Store Code in FAISS
     chunks = tokenize_and_chunk_code(generated_code)
     store_code_in_faiss(chunks)