Add live logs and API endpoints to the web interface.

app.py

@@ -1,28 +1,20 @@
 import os
-from flask import Flask, render_template
+import subprocess
+from flask import Flask, render_template, Response
 
 app = Flask(__name__)
 
 @app.route("/")
 def index():
-    logs = ""
-    with open("logs.txt", "r") as f:
-        logs = f.read()
-    return render_template("index.html", logs=logs)
+    return render_template("index.html", logs_url="/logs", api_url="/api/generate")
 
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=5000, debug=True)
-
-app = Flask(__name__)
-
-@app.route("/")
-def index():
-    return render_template("index.html", logs_url="/logs")
+@app.route("/logs")
+def logs():
+    def stream_logs():
+        process = subprocess.Popen(["docker-compose", "logs", "-f", "ollama"], stdout=subprocess.PIPE)
+        for line in iter(process.stdout.readline, b''):
+            yield line.decode('utf-8')
+    return Response(stream_logs(), mimetype='text/plain')
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=5000, debug=True)
-@app.route("/logs")
-def logs():
-    with open("logs.txt", "r") as f:
-        logs = f.read()
-    return logs

templates/index.html

@@ -15,6 +15,8 @@
         <li><code>/translate</code>: Translate text</li>
         <li><code>/summarize</code>: Summarize text</li>
         <li><code>/question</code>: Answer a question</li>
+        <li><code>/completion</code>: Complete text</li>
+        <li><code>/embedding</code>: Get embeddings for text</li>
     </ul>
 </body>
 </html>