diff --git a/Prompt_Studio/src/components/CinePromptStudio.jsx b/Prompt_Studio/src/components/CinePromptStudio.jsx
index 7bf8806..b422f16 100644
--- a/Prompt_Studio/src/components/CinePromptStudio.jsx
+++ b/Prompt_Studio/src/components/CinePromptStudio.jsx
@@ -1250,10 +1250,26 @@ STRICT OUTPUT FORMAT: Return ONLY the final visual description text. Do not use
 
             {/* Image Generation Panel */}
             <div className="bg-slate-900/50 border border-slate-800 rounded-xl p-6 space-y-4">
-              <h3 className="text-lg font-bold text-slate-200 flex items-center gap-2">
-                <Image className="w-5 h-5 text-cinema-gold" />
-                Generated Image
-              </h3>
+              <div className="flex items-center justify-between">
+                <h3 className="text-lg font-bold text-slate-200 flex items-center gap-2">
+                  <Image className="w-5 h-5 text-cinema-gold" />
+                  Generated Image
+                </h3>
+                {generatedImage && (
+                  <div className="flex items-center gap-3">
+                    <span className="text-xs px-2 py-1 rounded-full bg-amber-900/50 text-amber-400 border border-amber-800">
+                      Edit Mode: Next prompt will modify this image
+                    </span>
+                    <button
+                      onClick={resetImage}
+                      className="text-xs px-3 py-1 bg-slate-700 hover:bg-slate-600 text-slate-300 rounded-lg transition-all flex items-center gap-1"
+                    >
+                      <RefreshCw className="w-3 h-3" />
+                      Start Fresh
+                    </button>
+                  </div>
+                )}
+              </div>
 
               {/* Reference Images Upload */}
               <div className="space-y-2">
@@ -1365,7 +1381,7 @@ STRICT OUTPUT FORMAT: Return ONLY the final visual description text. Do not use
                   ) : (
                     <>
                       <Sparkles className="w-5 h-5" />
-                      <span>Generate Image</span>
+                      <span>{generatedImage ? 'Edit Image' : 'Generate Image'}</span>
                     </>
                   )}
                 </button>