Spaces:

1yahoo
/

Test

Sleeping

App Files Files Community

1yahoo commited on Jan 27

Commit

492ea5a

verified ·

1 Parent(s): 46570d8

Update app.py

Browse files

Files changed (1) hide show

app.py +82 -62

app.py CHANGED Viewed

@@ -7,17 +7,20 @@ from fastapi.middleware.cors import CORSMiddleware
 from openai import OpenAI
 from chromadb.utils import embedding_functions
-app = FastAPI(title="Neural OS - Deep Vault")
-# إعداد الذاكرة والعميل
 STORAGE_PATH = "/data/neural_memory" if os.path.exists("/data") else "./neural_memory"
 chroma_client = chromadb.PersistentClient(path=STORAGE_PATH)
 default_ef = embedding_functions.DefaultEmbeddingFunction()
-collection = chroma_client.get_or_create_collection(name="vault_v6", embedding_function=default_ef)
-client = OpenAI(base_url="https://router.huggingface.co/hf-inference/v1", api_key=os.getenv("HF_TOKEN"))
-# واجهة المستخدم الاحترافية (نسخة يوسف المفضلة)
 @app.get("/", response_class=HTMLResponse)
 async def get_ui():
     return """
@@ -26,30 +29,30 @@ async def get_ui():
     <head>
         <meta charset="UTF-8">
         <meta name="viewport" content="width=device-width, initial-scale=1.0">
-        <title>Deep Neural Vault | يوسف</title>
         <script src="https://cdn.tailwindcss.com"></script>
         <script src="https://unpkg.com/react@18/umd/react.production.min.js"></script>
         <script src="https://unpkg.com/react-dom@18/umd/react-dom.production.min.js"></script>
         <script src="https://unpkg.com/@babel/standalone/babel.min.js"></script>
         <link href="https://fonts.googleapis.com/css2?family=IBM+Plex+Sans+Arabic:wght@300;400;600&display=swap" rel="stylesheet">
         <style>
-            body { font-family: 'IBM Plex Sans Arabic', sans-serif; background: #020617; color: #f8fafc; overflow: hidden; }
-            .neural-grid { background-image: radial-gradient(#1e293b 1px, transparent 1px); background-size: 30px 30px; }
-            .glass-card { background: rgba(15, 23, 42, 0.7); backdrop-filter: blur(12px); border: 1px solid rgba(255,255,255,0.1); }
-            @keyframes pulse { 0%, 100% { opacity: 0.5; } 50% { opacity: 1; transform: scale(1.1); } }
-            .animate-neural { animation: pulse 2s infinite; }
         </style>
     </head>
-    <body class="neural-grid">
         <div id="root"></div>
         <script type="text/babel">
-            const { useState, useEffect, useRef } = React;
             function App() {
                 const [messages, setMessages] = useState([]);
                 const [input, setInput] = useState("");
-                const [status, setStatus] = useState("idle"); // idle, thinking, memory
-                const [files, setFiles] = useState([]);
                 const sendMessage = async () => {
                     if(!input.trim()) return;
@@ -62,7 +65,11 @@ async def get_ui():
                         const response = await fetch('/v1/chat/completions', {
                             method: 'POST',
                             headers: {'Content-Type': 'application/json'},
-                            body: JSON.stringify({messages: [...messages, userMsg]})
                         });
                         const reader = response.body.getReader();
@@ -72,65 +79,76 @@ async def get_ui():
                         while (true) {
                             const {done, value} = await reader.read();
                             if (done) break;
-                            const text = new TextDecoder().decode(value).replace(/data: /g, '').replace(/\\n\\n/g, '');
                             if(text.includes("[DONE]")) break;
                             botMsg.content += text;
                             setMessages(prev => [...prev.slice(0, -1), {...botMsg}]);
-                            setStatus("memory");
                         }
-                    } catch (e) { console.error(e); }
                     setStatus("idle");
                 };
                 return (
-                    <div className="flex h-screen overflow-hidden">
-                        {/* القائمة الجانبية (خزنة المعرفة) */}
-                        <aside className="w-80 glass-card p-6 flex flex-col gap-6 hidden md:flex">
-                            <div className="flex items-center gap-3">
-                                <div className="p-2 bg-indigo-500/20 rounded-lg text-indigo-400">🧬</div>
-                                <h1 className="text-xl font-bold bg-gradient-to-r from-indigo-400 to-cyan-400 bg-clip-text text-transparent">Neural Vault</h1>
                             </div>
-                            <div className="border-2 border-dashed border-slate-700 rounded-xl p-8 text-center hover:border-indigo-500 transition-colors cursor-pointer">
-                                <p className="text-sm text-slate-400">اسحب الملفات هنا (RAG)</p>
-                            </div>
-                            <div className="flex-1 overflow-y-auto space-y-2">
-                                <p className="text-xs text-slate-500 uppercase tracking-widest">المستندات المفهرسة</p>
-                                {files.length === 0 && <p className="text-sm text-slate-600 italic">الخزنة فارغة حالياً...</p>}
                             </div>
                         </aside>
-                        {/* منطقة الدردشة الرئيسية */}
-                        <main className="flex-1 flex flex-col relative">
-                            <header className="p-6 border-b border-white/5 flex justify-between items-center">
-                                <div className="flex items-center gap-4">
-                                    <div className={`w-3 h-3 rounded-full ${status === 'thinking' ? 'bg-purple-500 animate-pulse' : status === 'memory' ? 'bg-cyan-500 animate-neural' : 'bg-slate-700'}`}></div>
-                                    <span className="text-sm text-slate-400 font-mono">Status: {status}</span>
                                 </div>
                             </header>
-                            <div className="flex-1 overflow-y-auto p-6 space-y-6 scroll-smooth">
                                 {messages.map((m, i) => (
-                                    <div key={i} className={`flex ${m.role === 'user' ? 'justify-start' : 'justify-end animate-fade-in'}`}>
-                                        <div className={`max-w-[80%] p-4 rounded-2xl ${m.role === 'user' ? 'bg-slate-800 text-slate-200' : 'bg-indigo-500/10 border border-indigo-500/20 text-indigo-100'}`}>
                                             {m.content}
                                         </div>
                                     </div>
                                 ))}
                             </div>
-                            <div className="p-6 bg-slate-900/50 backdrop-blur-md">
-                                <div className="max-w-4xl mx-auto flex gap-4">
                                     <input
                                         value={input}
-                                        onChange={(e) => setInput(e.target.value)}
-                                        onKeyPress={(e) => e.key === 'Enter' && sendMessage()}
-                                        placeholder="تحدث مع الوعي الاصطناعي..."
-                                        className="flex-1 bg-slate-950/50 border border-white/10 rounded-xl px-6 py-4 focus:outline-none focus:border-indigo-500 transition-all shadow-2xl"
                                     />
-                                    <button onClick={sendMessage} className="bg-indigo-600 hover:bg-indigo-500 px-8 rounded-xl font-bold transition-all shadow-lg shadow-indigo-500/20">إرسال</button>
                                 </div>
                             </div>
                         </main>
@@ -145,33 +163,35 @@ async def get_ui():
     </html>
     """
-# المسارات الخلفية (Backend)
 @app.post("/v1/chat/completions")
 async def chat_endpoint(request: Request):
     body = await request.json()
     messages = body.get("messages", [])
-    # استرجاع المعرفة (RAG)
-    user_query = messages[-1]["content"] if messages else ""
-    results = collection.query(query_texts=[user_query], n_results=2)
-    context = "\\n".join(results['documents'][0]) if results['documents'] else ""
-    # حقن السياق
-    full_messages = [{"role": "system", "content": f"Use this memory context: {context}"}] + messages
     def stream_gen():
         response = client.chat.completions.create(
             model="huihui-ai/Qwen2.5-72B-Instruct-abliterated",
-            messages=full_messages,
             stream=True
         )
         for chunk in response:
             if chunk.choices[0].delta.content:
-                yield f"data: {chunk.choices[0].delta.content}\\n\\n"
-        yield "data: [DONE]\\n\\n"
     return StreamingResponse(stream_gen(), media_type="text/event-stream")
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)

 from openai import OpenAI
 from chromadb.utils import embedding_functions
+app = FastAPI()
+# إعداد الذاكرة
 STORAGE_PATH = "/data/neural_memory" if os.path.exists("/data") else "./neural_memory"
 chroma_client = chromadb.PersistentClient(path=STORAGE_PATH)
 default_ef = embedding_functions.DefaultEmbeddingFunction()
+collection = chroma_client.get_or_create_collection(name="vault_v7", embedding_function=default_ef)
+# العميل - تم تصحيح المسار
+client = OpenAI(
+    base_url="https://api-inference.huggingface.co/v1",
+    api_key=os.getenv("HF_TOKEN")
+)
 @app.get("/", response_class=HTMLResponse)
 async def get_ui():
     return """
     <head>
         <meta charset="UTF-8">
         <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>Deep Neural Vault</title>
         <script src="https://cdn.tailwindcss.com"></script>
         <script src="https://unpkg.com/react@18/umd/react.production.min.js"></script>
         <script src="https://unpkg.com/react-dom@18/umd/react-dom.production.min.js"></script>
         <script src="https://unpkg.com/@babel/standalone/babel.min.js"></script>
         <link href="https://fonts.googleapis.com/css2?family=IBM+Plex+Sans+Arabic:wght@300;400;600&display=swap" rel="stylesheet">
         <style>
+            body { font-family: 'IBM Plex Sans Arabic', sans-serif; background: #020617; color: #f8fafc; }
+            .glass { background: rgba(15, 23, 42, 0.8); backdrop-filter: blur(10px); border: 1px solid rgba(255,255,255,0.05); }
+            .custom-scrollbar::-webkit-scrollbar { width: 4px; }
+            .custom-scrollbar::-webkit-scrollbar-thumb { background: #334155; border-radius: 10px; }
         </style>
     </head>
+    <body>
         <div id="root"></div>
         <script type="text/babel">
+            const { useState, useEffect } = React;
             function App() {
                 const [messages, setMessages] = useState([]);
                 const [input, setInput] = useState("");
+                const [status, setStatus] = useState("idle");
+                const [isSidebarOpen, setSidebarOpen] = useState(false);
+                const [settings, setSettings] = useState({ temp: 0.7, tokens: 2048 });
                 const sendMessage = async () => {
                     if(!input.trim()) return;
                         const response = await fetch('/v1/chat/completions', {
                             method: 'POST',
                             headers: {'Content-Type': 'application/json'},
+                            body: JSON.stringify({
+                                messages: [...messages, userMsg],
+                                temperature: settings.temp,
+                                max_tokens: settings.tokens
+                            })
                         });
                         const reader = response.body.getReader();
                         while (true) {
                             const {done, value} = await reader.read();
                             if (done) break;
+                            const text = new TextDecoder().decode(value).replace(/data: /g, '');
                             if(text.includes("[DONE]")) break;
                             botMsg.content += text;
                             setMessages(prev => [...prev.slice(0, -1), {...botMsg}]);
                         }
+                    } catch (e) { console.error("Error:", e); }
+                    setStatus("idle");
+                };
+                const uploadFile = async (e) => {
+                    const file = e.target.files[0];
+                    if(!file) return;
+                    setStatus("thinking");
+                    const fd = new FormData();
+                    fd.append('file', file);
+                    await fetch('/upload', { method: 'POST', body: fd });
                     setStatus("idle");
+                    alert("تمت الفهرسة");
                 };
                 return (
+                    <div className="flex h-screen w-full overflow-hidden">
+                        {/* القائمة الجانبية - متجاوبة */}
+                        <aside className={`${isSidebarOpen ? 'translate-x-0' : 'translate-x-full'} fixed inset-y-0 right-0 z-50 w-72 glass p-6 transition-transform md:relative md:translate-x-0 flex flex-col gap-6`}>
+                            <div className="flex justify-between items-center">
+                                <h2 className="font-bold text-indigo-400">الإعدادات والذاكرة</h2>
+                                <button className="md:hidden" onClick={() => setSidebarOpen(false)}>✕</button>
                             </div>
+                            <div className="space-y-4">
+                                <label className="block text-xs text-slate-400 uppercase">الحرارة (الإبداع): {settings.temp}</label>
+                                <input type="range" min="0" max="2" step="0.1" value={settings.temp} onChange={e => setSettings({...settings, temp: parseFloat(e.target.value)})} className="w-full h-1 bg-slate-700 rounded-lg appearance-none cursor-pointer" />
+                                <label className="block text-xs text-slate-400 mt-4">رفع مستند (PDF/TXT)</label>
+                                <input type="file" onChange={uploadFile} className="block w-full text-xs text-slate-400 file:mr-4 file:py-2 file:px-4 file:rounded-full file:border-0 file:text-xs file:bg-indigo-600 file:text-white hover:file:bg-indigo-500 cursor-pointer"/>
                             </div>
                         </aside>
+                        {/* منطقة المحادثة */}
+                        <main className="flex-1 flex flex-col w-full relative">
+                            <header className="p-4 border-b border-white/5 flex justify-between items-center">
+                                <button className="md:hidden p-2" onClick={() => setSidebarOpen(true)}>☰</button>
+                                <div className="flex items-center gap-2">
+                                    <div className={`w-2 h-2 rounded-full ${status !== 'idle' ? 'bg-indigo-500 animate-pulse' : 'bg-slate-600'}`}></div>
+                                    <span className="text-xs font-mono">NEURAL_SYS: {status}</span>
                                 </div>
                             </header>
+                            <div className="flex-1 overflow-y-auto p-4 space-y-4 custom-scrollbar">
                                 {messages.map((m, i) => (
+                                    <div key={i} className={`flex ${m.role === 'user' ? 'justify-start' : 'justify-end'}`}>
+                                        <div className={`max-w-[90%] md:max-w-[70%] p-3 rounded-2xl text-sm ${m.role === 'user' ? 'bg-slate-800' : 'bg-indigo-900/30 border border-indigo-500/20'}`}>
                                             {m.content}
                                         </div>
                                     </div>
                                 ))}
                             </div>
+                            <div className="p-4 glass">
+                                <div className="max-w-4xl mx-auto flex gap-2">
                                     <input
                                         value={input}
+                                        onChange={e => setInput(e.target.value)}
+                                        onKeyDown={e => e.key === 'Enter' && sendMessage()}
+                                        placeholder="اكتب شيئاً..."
+                                        className="flex-1 bg-slate-900 border border-white/10 rounded-xl px-4 py-3 text-sm focus:outline-none focus:border-indigo-500"
                                     />
+                                    <button onClick={sendMessage} className="bg-indigo-600 p-3 rounded-xl hover:bg-indigo-500 transition-colors">
+                                        <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24"><path d="M5 12h14M12 5l7 7-7 7" strokeWidth="2" strokeLinecap="round" strokeLinejoin="round"/></svg>
+                                    </button>
                                 </div>
                             </div>
                         </main>
     </html>
     """
+@app.post("/upload")
+async def upload_file(background_tasks: BackgroundTasks, file: UploadFile = File(...)):
+    content = await file.read()
+    # هنا يتم استدعاء وظيفة الـ Ingest التي كتبتها سابقاً
+    text = content.decode("utf-8", errors="ignore")
+    collection.add(documents=[text], ids=[str(uuid.uuid4())])
+    return {"status": "ok"}
 @app.post("/v1/chat/completions")
 async def chat_endpoint(request: Request):
     body = await request.json()
     messages = body.get("messages", [])
+    temp = body.get("temperature", 0.7)
     def stream_gen():
+        # استخدام موديل بديل إذا استمر الخطأ في Qwen
         response = client.chat.completions.create(
             model="huihui-ai/Qwen2.5-72B-Instruct-abliterated",
+            messages=messages,
+            temperature=temp,
             stream=True
         )
         for chunk in response:
             if chunk.choices[0].delta.content:
+                yield f"data: {chunk.choices[0].delta.content}\n\n"
+        yield "data: [DONE]\n\n"
     return StreamingResponse(stream_gen(), media_type="text/event-stream")
 if __name__ == "__main__":
     import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)