Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+import streamlit as st
 import openai
 import os
 import tempfile
@@ -577,17 +577,27 @@ def execute_code_with_retry(code: str, max_attempts: int = 3) -> Tuple[bool, str
     
     return False, "Max attempts reached", None
 
-def process_request(instruction, files, urls_input):
-    """Main processing function with self-correction, supporting URLs."""
+def save_uploaded_files(uploaded_files):
+    """Save uploaded Streamlit files to temp dir and return paths."""
+    if not uploaded_files:
+        return []
+    temp_dir = tempfile.mkdtemp(prefix='uploaded_files_')
+    file_paths = []
+    for uploaded_file in uploaded_files:
+        if uploaded_file is not None:
+            file_path = os.path.join(temp_dir, uploaded_file.name or 'unnamed_file')
+            with open(file_path, 'wb') as f:
+                f.write(uploaded_file.getbuffer())
+            file_paths.append(file_path)
+    return file_paths
+
+def process_request(instruction, uploaded_files, urls_input):
+    """Main processing function with self-correction, supporting URLs. Adapted for Streamlit."""
     try:
         if not instruction.strip():
             return "لطفاً دستور را وارد کنید. (فایل‌ها و لینک‌ها اختیاری هستند)", None
         
-        file_paths = []
-        
-        # Handle uploaded files
-        if files:
-            file_paths = [f.name for f in files]
+        file_paths = save_uploaded_files(uploaded_files)
         
         # Handle URLs: download to temp dir
         if urls_input and urls_input.strip():
@@ -601,8 +611,6 @@ def process_request(instruction, files, urls_input):
             file_paths.extend(downloaded_paths)
             print(f"Downloaded {len(downloaded_paths)} files from URLs")
         
-        # Clean up note: Temp dirs will be cleaned on shutdown or manually if needed
-        
         # Track errors for learning
         previous_errors = []
         generated_codes = []
@@ -665,11 +673,13 @@ def process_request(instruction, files, urls_input):
         print(error_msg)
         return error_msg, None
 
-# Gradio Interface
-with gr.Blocks(title="AI File Processor - Self Correcting") as demo:
-    gr.Markdown("""
-    # 🤖 AI File Processor - Self Correcting Edition
+# Streamlit Interface
+def main():
+    st.set_page_config(page_title="AI File Processor - Self Correcting", layout="wide")
     
+    st.title("🤖 AI File Processor - Self Correcting Edition")
+    
+    st.markdown("""
     این سیستم می‌تواند:
     - کد Python تولید کند
     - خطاها را تشخیص و تحلیل کند
@@ -684,40 +694,55 @@ with gr.Blocks(title="AI File Processor - Self Correcting") as demo:
     **نکته:** می‌توانید فایل‌ها را آپلود کنید یا لینک‌های فایل را (جدا شده با کاما) وارد کنید.
     """)
     
-    with gr.Row():
-        instruction = gr.Textbox(
+    col1, col2 = st.columns(2)
+    
+    with col1:
+        instruction = st.text_area(
             label="دستور", 
-            lines=2, 
+            height=100,
             placeholder="مثال: یک نمودار دایره‌ای از داده‌های فروش بکش",
-            rtl=True
+            help="دستور خود را به فارسی یا انگلیسی وارد کنید."
         )
     
-    with gr.Row():
-        files = gr.File(file_count="multiple", label="فایل‌های آپلود شده (اختیاری)")
-        urls = gr.Textbox(
+    with col2:
+        uploaded_files = st.file_uploader("فایل‌های آپلود شده (اختیاری)", type=['csv', 'xlsx', 'jpg', 'png', 'txt'], accept_multiple_files=True)
+        urls_input = st.text_input(
             label="لینک فایل‌ها (جدا با کاما، اختیاری)", 
             placeholder="https://example.com/file1.csv, https://example.com/file2.jpg",
-            info="لینک‌ها دانلود خواهند شد و حجم فایل چک نمی‌شود."
+            help="لینک‌ها دانلود خواهند شد."
         )
     
-    btn = gr.Button("🚀 اجرا", variant="primary")
-    
-    with gr.Row():
-        output = gr.Textbox(label="نتیجه", lines=15)
-        file_out = gr.File(label="فایل خروجی (در صورت تولید)")
-    
-    # Examples
-    gr.Examples(
-        examples=[
-            ["یک فایل اکسل با 100 محصول فروشگاهی شامل نام، قیمت و موجودی بساز", None, None],
-            ["یک نمودار میله‌ای از داده‌های تصادفی رسم کن", None, None],
-            ["یک تصویر 500x500 پیکسل با رنگ‌های تصادفی بساز", None, None],
-            ["این فایل را به فرمت JSON تبدیل کن", None, "https://example.com/sample.csv"],
-        ],
-        inputs=[instruction, files, urls],
-    )
+    # Examples as selectbox
+    examples = [
+        ("یک فایل اکسل با 100 محصول فروشگاهی شامل نام، قیمت و موجودی بساز", None, None),
+        ("یک نمودار میله‌ای از داده‌های تصادفی رسم کن", None, None),
+        ("یک تصویر 500x500 پیکسل با رنگ‌های تصادفی بساز", None, None),
+        ("این فایل را به فرمت JSON تبدیل کن", None, "https://example.com/sample.csv"),
+    ]
+    
+    selected_example = st.selectbox("انتخاب مثال:", ["هیچ‌کدام"] + [ex[0] for ex in examples])
+    if selected_example != "هیچ‌کدام":
+        idx = [ex[0] for ex in examples].index(selected_example)
+        instruction, _, urls_input = examples[idx]  # Note: files can't be pre-set easily in Streamlit
     
-    btn.click(fn=process_request, inputs=[instruction, files, urls], outputs=[output, file_out])
+    if st.button("🚀 اجرا", type="primary"):
+        with st.spinner("در حال پردازش..."):
+            result_text, output_file_path = process_request(instruction, uploaded_files, urls_input)
+            
+            st.subheader("نتیجه")
+            st.text_area("خروجی:", value=result_text, height=300, key="result")
+            
+            if output_file_path and os.path.exists(output_file_path):
+                with open(output_file_path, "rb") as file:
+                    st.download_button(
+                        label="دانلود فایل خروجی",
+                        data=file.read(),
+                        file_name=os.path.basename(output_file_path),
+                        mime="application/octet-stream"
+                    )
+                # Clean up after download (optional, but for Spaces it's fine to leave temp)
+            else:
+                st.info("هیچ فایل خروجی تولید نشد.")
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    main()