Update app.py

app.py

@@ -1,73 +1,17 @@
 import gradio as gr
 import openai
 import os
-from tempfile import NamedTemporaryFile
 
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-
-def process_files(xml_file, tcpou_file):
-    try:
-        with open(xml_file, 'r') as f:
-            xml_content = f.read()
-        
-        with open(tcpou_file, 'r') as f:
-            tcpou_content = f.read()
-
-    except Exception as e:
-        return f"Error reading files: {str(e)}", None
-
-    prompt = f"""Analyze these PLC development files:
-
-    XML File Content:
-    {xml_content}
-
-    TcPOU Code File Content:
-    {tcpou_content}
-
-    Provide a detailed analysis of the code structure, functionality, 
-    and any potential issues or improvements. Also include documentation suggestions.
-    """
-
-    try:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=[
-                {"role": "system", "content": "You are an industrial automation expert analyzing PLC code."},
-                {"role": "user", "content": prompt}
-            ]
-        )
-        analysis = response.choices[0].message.content
-    except Exception as e:
-        return f"API Error: {str(e)}", None
-
-    with NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
-        f.write(analysis)
-        temp_path = f.name
-
-    return analysis, temp_path
-
-with gr.Blocks() as demo:
-    gr.Markdown("# PLC Code Analyzer 🤖")
-    
-    with gr.Row():
-        with gr.Column():
-            xml_input = gr.File(label="Upload XML File", type="filepath")
-            tcpou_input = gr.File(label="Upload TcPOU File", type="filepath")
-            submit_btn = gr.Button("Analyze Files")
-        
-        with gr.Column():
-            output_text = gr.Textbox(label="Analysis Result", interactive=False)
-            download_output = gr.File(label="Download Result", interactive=False)
-
-    submit_btn.click(
-        fn=process_files,
-        inputs=[xml_input, tcpou_input],
-        outputs=[output_text, download_output]
+def analyze(xml, code):
+    prompt = f"XML: {xml}\nCode: {code}\nAnalyze:"
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[{"role": "user", "content": prompt}]
     )
+    return response.choices[0].message.content
 
-if __name__ == "__main__":
-    if not OPENAI_API_KEY:
-        gr.Error("OpenAI API key not configured! Add it in Space Settings → Secrets")
-    
-    openai.api_key = OPENAI_API_KEY
-    demo.launch(concurrency_count=20)  # Corrected line
+gr.Interface(
+    fn=analyze,
+    inputs=[gr.File(), gr.File()],
+    outputs=[gr.Textbox(), gr.File()]
+).launch()