make the image display properly

DefangLabs · Nov 28, 2024 · 7c9c778 · 7c9c778
1 parent d0449d9
commit 7c9c778
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/app/app.py b/app/app.py
@@ -1,7 +1,7 @@
 from flask import Flask, request, jsonify, render_template, Response, stream_with_context
 from rag_system import rag_system
 import subprocess
-app = Flask(__name__)
+app = Flask(__name__, static_folder='templates/images')
 
 @app.route('/', methods=['GET', 'POST'])
 def index():

diff --git a/app/requirements.txt b/app/requirements.txt
@@ -1,5 +1,5 @@
 Flask==2.0.1
-Werkzeug==3.0.6
+Werkzeug==2.2.2
 scikit-learn==0.24.2
 numpy==1.21.0
 sentence-transformers==2.1.0

diff --git a/app/templates/imgs/defang_icon.svg → app/templates/images/defang_icon.svg b/app/templates/imgs/defang_icon.svg → app/templates/images/defang_icon.svg
diff --git a/app/templates/index.html b/app/templates/index.html
@@ -148,7 +148,7 @@
 <body>
     <div class="chat-container">
         <div class="img-container">
-            <img class="icon" src="imgs/defang_icon.svg"/>
+            <img class="icon" src="{{ url_for('static', filename='defang_icon.svg') }}" alt="defang_icon"/>
         </div>
         <h2>Ask Defang</h2>
         <div id="chat-box" class="chat-box"></div>